package com.castle.fortress.admin.ai.service.impl;

import cn.hutool.core.util.StrUtil;

import com.castle.fortress.common.utils.ConvertUtil;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.castle.fortress.admin.ai.entity.AiGroupEntity;
import com.castle.fortress.admin.ai.dto.AiGroupDto;
import com.castle.fortress.admin.ai.mapper.AiGroupMapper;
import com.castle.fortress.admin.ai.service.AiGroupService;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import java.util.Map;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import java.util.List;

/**
 * ai 组 服务实现类
 *
 * @author 
 * @since 2024-03-09
 */
@Service
public class AiGroupServiceImpl extends ServiceImpl<AiGroupMapper, AiGroupEntity> implements AiGroupService {
    /**
    * 获取查询条件
    * @param aiGroupDto
    * @return
    */
    public QueryWrapper<AiGroupEntity> getWrapper(AiGroupDto aiGroupDto){
        QueryWrapper<AiGroupEntity> wrapper= new QueryWrapper();
        if(aiGroupDto != null){
            AiGroupEntity aiGroupEntity = ConvertUtil.transformObj(aiGroupDto,AiGroupEntity.class);
            wrapper.like(aiGroupEntity.getId() != null,"id",aiGroupEntity.getId());
            wrapper.like(aiGroupEntity.getUserId() != null,"user_id",aiGroupEntity.getUserId());
            wrapper.like(StrUtil.isNotEmpty(aiGroupEntity.getName()),"name",aiGroupEntity.getName());
            wrapper.like(aiGroupEntity.getCreateTime() != null,"create_time",aiGroupEntity.getCreateTime());
            wrapper.like(aiGroupEntity.getCreateUser() != null,"create_user",aiGroupEntity.getCreateUser());
            wrapper.like(aiGroupEntity.getUpdateTime() != null,"update_time",aiGroupEntity.getUpdateTime());
        }
        return wrapper;
    }


    @Override
    public IPage<AiGroupDto> pageAiGroup(Page<AiGroupDto> page, AiGroupDto aiGroupDto) {
        QueryWrapper<AiGroupEntity> wrapper = getWrapper(aiGroupDto);
        Page<AiGroupEntity> pageEntity = new Page(page.getCurrent(), page.getSize());
        Page<AiGroupEntity> aiGroupPage=baseMapper.selectPage(pageEntity,wrapper);
        Page<AiGroupDto> pageDto = new Page(aiGroupPage.getCurrent(), aiGroupPage.getSize(),aiGroupPage.getTotal());
        pageDto.setRecords(ConvertUtil.transformObjList(aiGroupPage.getRecords(),AiGroupDto.class));
        return pageDto;
    }


    @Override
    public List<AiGroupDto> listAiGroup(AiGroupDto aiGroupDto){
        QueryWrapper<AiGroupEntity> wrapper = getWrapper(aiGroupDto);
        wrapper.orderByDesc("update_time");
        List<AiGroupEntity> list = baseMapper.selectList(wrapper);
        return ConvertUtil.transformObjList(list,AiGroupDto.class);
    }
}

